from ldm.devices.devices import choose_autocast_device, choose_torch_device